package org.simpleframework.util;

import java.io.File;
import java.io.FileFilter;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.net.URL;
import java.util.HashSet;
import java.util.Set;

/**
 * @author 江黎
 * @date 2022-01-29
 */
public class ClassUtil {

    /**
     * 文件协议
     */
    private static final String FILE_PROTOCOL = "file";
    /**
     * 字节码文件
     */
    private static final String CLASS_SUFFIX = ".class";

    /**
     * 获取当前包所有类的集合
     *
     * @param packageName
     * @return
     */
    public static Set<Class<?>> extractPackageClass(String packageName) {
        // 1.获取到类的加载器
        ClassLoader classLoader = getClassLoader();
        // 2.通过类加载器获取到加载的资源信息
        URL url = classLoader.getResource(packageName.replace(".", "/"));
        if (url == null) {
            System.out.println("unable to retrieve anything from package " + packageName);
            return null;
        }
        // 3.根据不同的资源类型,采用不同的方式获取资源的集合
        Set<Class<?>> classSet = null;
        // 过滤出文件类型的资源
        if (url.getProtocol().equalsIgnoreCase(FILE_PROTOCOL)) {
            classSet = new HashSet<>();
            File packageDirectory = new File(url.getPath());
            extractClassFile(classSet, packageDirectory, packageName);
        }

        // else { } TODO 此处还可以加入对其他类型资源的处理


        return classSet;
    }

    /**
     * 递归获取目标package里面的所有class文件(包括子package)
     *
     * @param classSet    装载目标类的集合
     * @param fileSource  文件或者目录(扫包的目录)
     * @param packageName 包名
     */
    private static void extractClassFile(Set<Class<?>> classSet, File fileSource, String packageName) {
        // 不是一个目录则直接结束递归方法
        if (!fileSource.isDirectory()) {
            return;
        }

        // 如果是一个文件夹,则调用listFiles方法获取文件夹下的文件或文件夹
        File[] files = fileSource.listFiles(new FileFilter() {
            @Override
            public boolean accept(File file) {
                if (file.isDirectory()) {
                    return true;
                } else {
                    // 获取文件的绝对路径
                    String absolutePath = file.getAbsolutePath();
                    // 判断是不是字节码文件
                    if (absolutePath.endsWith(CLASS_SUFFIX)) {
                        // 若是class文件,则放入classSet
                        addToClassSet(absolutePath);
                    }
                }
                return false;
            }

            // 根据class文件的绝对路径,获取并生成class对象,并放入classSet中
            private void addToClassSet(String absolutePath) {
                // 1.从class文件的绝对路径里提取出包含了package的类名
                absolutePath = absolutePath.replace(File.separator, ".");
                // 去掉项目路径
                String className = absolutePath.substring(absolutePath.indexOf(packageName));
                // 去掉.class后缀
                className = className.substring(0, className.lastIndexOf("."));
                // 2.通过反射获取对应的Class对象并放入到classSet中
                try {
                    Class<?> targetClass = Class.forName(className);
                    classSet.add(targetClass);

                } catch (ClassNotFoundException e) {
                    e.printStackTrace();
                    throw new RuntimeException(e);
                }
            }
        });

        if (files != null && files.length != 0) {
            for (File file : files) {
                // 递归调用
                extractClassFile(classSet, file, packageName);
            }
        }

    }

    /**
     * 获取ClassLoader
     *
     * @return
     */
    private static ClassLoader getClassLoader() {
        return Thread.currentThread().getContextClassLoader();
    }

    /**
     * 实例化class
     */
    public static <T> T newInstance(Class<?> clazz) {
        try {
            Constructor<?> constructor = clazz.getDeclaredConstructor();
            constructor.setAccessible(true);
            return (T) constructor.newInstance();
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    /**
     * 设置类的属性值
     */
    public static void setField(Object target, Field field, Object value) {
        field.setAccessible(true);
        try {
            field.set(target, value);
        } catch (IllegalAccessException e) {
            System.out.println("setField error " + e.getMessage());
            throw new RuntimeException(e);
        }
    }

//    public static void main(String[] args) {
//        Set<Class<?>> classSet = extractPackageClass("com.imooc.entity");
//        System.out.println();
//    }
}
